Tratamento dos dados

In [ ]:

import numpy as np
import os
import zipfile
import cv2
import tensorflow as tf
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import warnings

import numpy as np import os import zipfile import cv2 import tensorflow as tf import pandas as pd import seaborn as sns import matplotlib.pyplot as plt import warnings

Armazenando todos os arquivos em uma lista

In [7]:

diretorio = 'data/terra'
arquivos = [os.path.join(diretorio, f) for f in sorted(os.listdir(diretorio))]
print(arquivos)

diretorio = 'data/terra' arquivos = [os.path.join(diretorio, f) for f in sorted(os.listdir(diretorio))] print(arquivos)

['data/terra/-0.625°N, 102°E.jpg', 'data/terra/-1.75°N, 309.375°E.jpg', 'data/terra/-2°N, 305.125°E.jpg', 'data/terra/-2°N, 306.625°E.jpg', 'data/terra/-3.625°N, 15°E.jpg', 'data/terra/-3°N, 299.75°E.jpg', 'data/terra/-4.125°N, 286.75°E.jpg', 'data/terra/-5.125°N, 298.75°E.jpg', 'data/terra/-8.125°N, 295.875°E.jpg', 'data/terra/0.125°N, 113.5°E.jpg', 'data/terra/0.25°N, 109.375°E.jpg', 'data/terra/15°N, 270°E.jpg', 'data/terra/20.75°N, 283.625°E.jpg', 'data/terra/25°N, 281.375°E.jpg', 'data/terra/28.375°N, 98.75°E.jpg', 'data/terra/36.27°N, 138.57°E.jpg', 'data/terra/36.5°N, 139.5°E.jpg', 'data/terra/4.875°N, 304.75°E.jpg', 'data/terra/44.25°N, 281°E.jpg', 'data/terra/44.75°N, 273.125°E.jpg', 'data/terra/49.5°N, 139.625°E.jpg', 'data/terra/4°N, 286.75°E.jpg', 'data/terra/53.707°N, 96.848°E.jpg', 'data/terra/5°N, 18.625°E.jpg', 'data/terra/6°N, 359.875°E.jpg', 'data/terra/9.75°N, 38.625°E.jpg', 'data/terra/9°N, 349.25°E.jpg']

Verificando tamanho da imagem

In [14]:

imagem = cv2.imread('data/terra/-0.625°N, 102°E.jpg')
imagem.shape

imagem = cv2.imread('data/terra/-0.625°N, 102°E.jpg') imagem.shape

Out[14]:

(527, 1585, 3)

In [9]:

largura, altura = 1585, 527

largura, altura = 1585, 527

Out[9]:

(527, 1585, 3)

Passando imagens de treino para cores de cinza

In [13]:

imagens = []
classes = []
for imagem_caminho in arquivos:
  try:
    imagem = cv2.imread(imagem_caminho)
    (H, W) = imagem.shape[:2]
  except:
    continue

  imagem = cv2.resize(imagem, (largura, altura))
  imagem = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY)

  plt.imshow(imagem)
  plt.show()
    
  imagem = imagem.ravel()
  imagens.append(imagem)
  classes.append('terra')

imagens = [] classes = [] for imagem_caminho in arquivos: try: imagem = cv2.imread(imagem_caminho) (H, W) = imagem.shape[:2] except: continue imagem = cv2.resize(imagem, (largura, altura)) imagem = cv2.cvtColor(imagem, cv2.COLOR_BGR2GRAY) plt.imshow(imagem) plt.show() imagem = imagem.ravel() imagens.append(imagem) classes.append('terra')

In [ ]:

In [ ]: